from openpyxl import load_workbook


# 数据驱动，通过这种方式  可以提高测试的复用性，减少重复测试的次数
class ReadExcel:
    def get_excel_data(self, ExcelName, sheet):
        # 打开表格
        wb = load_workbook(f"../woniuExcel/{ExcelName}.xlsx")
        ws = wb[sheet]
        rows = ws.max_row
        cols = ws.max_column

        info_list = []

        for r in range(2, rows + 1):
            # 定义一个列表 用来保存一个的信息
            lines = []
            for c in range(2, cols + 1):
                lines.append(ws.cell(r, c).value)
            info_list.append(lines)

        return info_list


if __name__ == '__main__':
    r = ReadExcel()
    e = r.get_excel_data()
    print(e)
